import sys
import os
sys.path.append(os.path.abspath("."))
from utils import *

import sys
import os
sys.path.append(os.path.abspath("."))
from utils import *
from collections import Counter

loader = DataLoader(
    meta_path="./data/meta_0db_id_00.json", 
    mode="logmel", 
    n_flag="abnormal",
    p_flag="normal"    
)

model = ASCNN()
for bx, by in loader:
    print(bx.shape, by.shape, bx.dtype, by.dtype)
    bottle, decode = model(bx)
    print(bottle.shape, decode.shape)
    break
